# -*- coding: utf-8 -*-'''非线性层'''
import torch

input = torch.tensor([[[[1, -0.5],
                        [-1,   3]]]])

relu1 = torch.nn.ReLU(False) # inplace=True ： 对源数据进行操作
relu2 = torch.nn.ReLU(True)

print(relu1(input))
print(relu2(input))

sigmoid = torch.nn.Sigmoid()

print(sigmoid(input))

